package org.yzhzc.friend.manager;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.collection.CollectionUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.github.pagehelper.PageHelper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.yzhzc.common.core.constants.CacheConstants;
import org.yzhzc.common.core.constants.Constants;
import org.yzhzc.common.core.enums.ExamListType;
import org.yzhzc.common.core.enums.ResultCode;
import org.yzhzc.common.redis.service.RedisService;
import org.yzhzc.common.security.exception.ServiceException;
import org.yzhzc.friend.domain.exam.dto.ExamQueryDTO;
import org.yzhzc.friend.domain.exam.dto.ExamRankDTO;
import org.yzhzc.friend.domain.exam.entity.Exam;
import org.yzhzc.friend.domain.exam.entity.ExamQuestion;
import org.yzhzc.friend.domain.exam.vo.ExamRankVO;
import org.yzhzc.friend.domain.exam.vo.ExamVO;
import org.yzhzc.friend.domain.user.entity.UserExam;
import org.yzhzc.friend.mapper.exam.ExamMapper;
import org.yzhzc.friend.mapper.exam.ExamQuestionMapper;
import org.yzhzc.friend.mapper.user.UserExamMapper;

import java.time.LocalDateTime;
import java.time.temporal.ChronoUnit;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

/**
 * 竞赛缓存管理类，负责竞赛相关数据的缓存操作。
 * 包括竞赛列表、竞赛题目列表、竞赛排名列表等。
 */
@Component
public class ExamCacheManager {

    @Autowired
    private ExamMapper examMapper; // 竞赛数据访问接口

    @Autowired
    private ExamQuestionMapper examQuestionMapper; // 竞赛题目数据访问接口

    @Autowired
    private UserExamMapper userExamMapper; // 用户竞赛数据访问接口

    @Autowired
    private RedisService redisService; // Redis 服务接口

    /**
     * 获取竞赛列表的大小。
     *
     * @param examListType 竞赛列表类型（未完成、历史、用户竞赛等）
     * @param userId       用户ID
     * @return 竞赛列表的大小
     */
    public Long getListSize(Integer examListType, Long userId) {
        String examListKey = getExamListKey(examListType, userId);
        return redisService.getListSize(examListKey);
    }

    /**
     * 获取竞赛题目列表的大小。
     *
     * @param examId 竞赛ID
     * @return 竞赛题目列表的大小
     */
    public Long getExamQuestionListSize(Long examId) {
        String examQuestionListKey = getExamQuestionListKey(examId);
        return redisService.getListSize(examQuestionListKey);
    }

    /**
     * 获取竞赛排名列表的大小。
     *
     * @param examId 竞赛ID
     * @return 竞赛排名列表的大小
     */
    public Long getRankListSize(Long examId) {
        return redisService.getListSize(getExamRankListKey(examId));
    }

    /**
     * 根据竞赛查询条件和用户ID，从 Redis 缓存中获取用户参加的竞赛ID列表，并组装成竞赛VO列表。
     * 如果 Redis 中没有数据或数据不完整，则从数据库中查询数据并刷新缓存。
     *
     * @param examQueryDTO 竞赛查询条件，包含分页信息和竞赛类型
     * @param userId       用户ID（用于生成用户竞赛列表的缓存键）
     * @return 竞赛VO列表
     */
    public List<ExamVO> getExamVOList(ExamQueryDTO examQueryDTO, Long userId) {
        // 计算分页的起始位置和结束位置
        int start = (examQueryDTO.getPageNum() - 1) * examQueryDTO.getPageSize();
        int end = start + examQueryDTO.getPageSize() - 1; // 下标需要 -1

        // 从 Redis 中获取指定范围内用户参加的 竞赛ID 列表
        String examListKey = getExamListKey(examQueryDTO.getType(), userId);
        List<Long> examIdList = redisService.getCacheListByRange(examListKey, start, end, Long.class);

        // 根据竞赛ID列表组装竞赛VO列表
        List<ExamVO> examVOList = assembleExamVOList(examIdList);

        // 检查组装后的竞赛VO列表是否为空
        if (CollectionUtil.isEmpty(examVOList)) {
            // 如果 Redis 中数据可能有问题，从数据库中查询数据
            examVOList = getExamListByDB(examQueryDTO, userId);

            // 刷新缓存，确保下次查询时缓存中有数据
            refreshCache(examQueryDTO.getType(), userId);
        }

        // 返回竞赛VO列表
        return examVOList;
    }

    /**
     * 获取竞赛排名列表。
     *
     * @param examRankDTO 竞赛排名查询条件
     * @return 竞赛排名列表
     */
    public List<ExamRankVO> getExamRankList(ExamRankDTO examRankDTO) {
        int start = (examRankDTO.getPageNum() - 1) * examRankDTO.getPageSize();
        int end = start + examRankDTO.getPageSize() - 1; // 下标需要 -1
        return redisService.getCacheListByRange(getExamRankListKey(examRankDTO.getExamId()), start, end, ExamRankVO.class);
    }

    /**
     * 获取用户的所有竞赛列表。
     *
     * @param userId 用户ID
     * @return 用户竞赛ID列表
     */
    public List<Long> getAllUserExamList(Long userId) {
        String examListKey = CacheConstants.USER_EXAM_LIST + userId;
        List<Long> userExamIdList = redisService.getCacheListByRange(examListKey, 0, -1, Long.class);
        if (CollectionUtil.isNotEmpty(userExamIdList)) {
            return userExamIdList;
        }
        List<UserExam> userExamList =
                userExamMapper.selectList(new LambdaQueryWrapper<UserExam>().eq(UserExam::getUserId, userId));
        if (CollectionUtil.isEmpty(userExamList)) {
            return null;
        }
        refreshCache(ExamListType.USER_EXAM_LIST.getValue(), userId);
        return userExamList.stream().map(UserExam::getExamId).collect(Collectors.toList());
    }

    /**
     * 添加用户竞赛缓存。
     *
     * @param userId 用户ID
     * @param examId 竞赛ID
     */
    public void addUserExamCache(Long userId, Long examId) {
        String userExamListKey = getUserExamListKey(userId);
        redisService.leftPushForList(userExamListKey, examId);
    }

    /**
     * 获取竞赛的第一个题目ID。
     *
     * @param examId 竞赛ID
     * @return 第一个题目ID
     */
    public Long getFirstQuestion(Long examId) {
        return redisService.indexForList(getExamQuestionListKey(examId), 0, Long.class);
    }

    /**
     * 获取当前题目的上一个题目ID。
     *
     * @param examId     竞赛ID
     * @param questionId 当前题目ID
     * @return 上一个题目ID
     */
    public Long preQuestion(Long examId, Long questionId) {
        Long index = redisService.indexOfForList(getExamQuestionListKey(examId), questionId);
        if (index == 0) {
            throw new ServiceException(ResultCode.FAILED_FIRST_QUESTION);
        }
        return redisService.indexForList(getExamQuestionListKey(examId), index - 1, Long.class);
    }

    /**
     * 获取当前题目的下一个题目ID。
     *
     * @param examId     竞赛ID
     * @param questionId 当前题目ID
     * @return 下一个题目ID
     */
    public Long nextQuestion(Long examId, Long questionId) {
        Long index = redisService.indexOfForList(getExamQuestionListKey(examId), questionId);
        long lastIndex = getExamQuestionListSize(examId) - 1;
        if (index == lastIndex) {
            throw new ServiceException(ResultCode.FAILED_LAST_QUESTION);
        }
        return redisService.indexForList(getExamQuestionListKey(examId), index + 1, Long.class);
    }

    /**
     * 刷新缓存逻辑。
     * 根据竞赛列表类型和用户ID，从数据库中查询最新的竞赛数据，并更新到 Redis 缓存中。
     *
     * @param examListType 竞赛列表类型（未完成竞赛、历史竞赛、用户竞赛）
     * @param userId       用户ID（用于生成用户竞赛列表的缓存键）
     */
    public void refreshCache(Integer examListType, Long userId) {
        // 初始化竞赛列表
        List<Exam> examList = new ArrayList<>();

        // 根据竞赛列表类型查询对应的竞赛数据
        if (ExamListType.EXAM_UN_FINISH_LIST.getValue().equals(examListType)) {
            // 查询未完成的竞赛列表
            examList = examMapper.selectList(new LambdaQueryWrapper<Exam>()
                    .select(Exam::getExamId, Exam::getTitle, Exam::getStartTime, Exam::getEndTime)
                    .gt(Exam::getEndTime, LocalDateTime.now()) // 结束时间大于当前时间
                    .eq(Exam::getStatus, Constants.TRUE) // 状态为启用
                    .orderByDesc(Exam::getCreateTime)); // 按创建时间降序排序
        } else if (ExamListType.EXAM_HISTORY_LIST.getValue().equals(examListType)) {
            // 查询历史竞赛列表
            examList = examMapper.selectList(new LambdaQueryWrapper<Exam>()
                    .select(Exam::getExamId, Exam::getTitle, Exam::getStartTime, Exam::getEndTime)
                    .le(Exam::getEndTime, LocalDateTime.now()) // 结束时间小于等于当前时间
                    .eq(Exam::getStatus, Constants.TRUE) // 状态为启用
                    .orderByDesc(Exam::getCreateTime)); // 按创建时间降序排序
        } else if (ExamListType.USER_EXAM_LIST.getValue().equals(examListType)) {
            // 查询用户的竞赛列表
            List<ExamVO> examVOList = userExamMapper.selectUserExamList(userId);
            examList = BeanUtil.copyToList(examVOList, Exam.class); // 将 ExamVO 转换为 Exam
        }

        // 如果查询结果为空，直接返回
        if (CollectionUtil.isEmpty(examList)) {
            return;
        }

        Map<String, Exam> examMap = new HashMap<>();    // 将竞赛数据存储到 Map 中，用于刷新详情缓存
        List<Long> examIdList = new ArrayList<>();      // 存储竞赛ID列表，用于刷新列表缓存

        for (Exam exam : examList) {
            // 生成竞赛详情的缓存键，并将竞赛数据存入 Map
            examMap.put(getDetailKey(exam.getExamId()), exam);
            // 将竞赛ID添加到列表中
            examIdList.add(exam.getExamId());
        }

        // 刷新竞赛详情缓存
        redisService.multiSet(examMap);

        // 删除旧的竞赛列表缓存
        redisService.deleteObject(getExamListKey(examListType, userId));

        // 刷新竞赛列表缓存
        redisService.rightPushAll(getExamListKey(examListType, userId), examIdList);
    }

    /**
     * 刷新竞赛题目缓存。
     *
     * @param examId 竞赛ID
     */
    public void refreshExamQuestionCache(Long examId) {
        List<ExamQuestion> examQuestionList = examQuestionMapper.selectList(new LambdaQueryWrapper<ExamQuestion>()
                .select(ExamQuestion::getQuestionId)
                .eq(ExamQuestion::getExamId, examId)
                .orderByAsc(ExamQuestion::getQuestionOrder));
        if (CollectionUtil.isEmpty(examQuestionList)) {
            return;
        }
        List<Long> examQuestionIdList = examQuestionList.stream().map(ExamQuestion::getQuestionId).toList();
        redisService.rightPushAll(getExamQuestionListKey(examId), examQuestionIdList);
        // 节省 Redis 缓存资源
        long seconds = ChronoUnit.SECONDS.between(LocalDateTime.now(),
                LocalDateTime.now().plusDays(1).withHour(0).withMinute(0).withSecond(0).withNano(0));
        redisService.expire(getExamQuestionListKey(examId), seconds, TimeUnit.SECONDS);
    }

    /**
     * 刷新竞赛排名缓存。
     *
     * @param examId 竞赛ID
     */
    public void refreshExamRankCache(Long examId) {
        List<ExamRankVO> examRankVOList = userExamMapper.selectExamRankList(examId);
        if (CollectionUtil.isEmpty(examRankVOList)) {
            return;
        }
        redisService.rightPushAll(getExamRankListKey(examId), examRankVOList);
    }

    /**
     * 从数据库中获取竞赛列表。
     *
     * @param examQueryDTO 竞赛查询条件
     * @param userId       用户ID
     * @return 竞赛VO列表
     */
    private List<ExamVO> getExamListByDB(ExamQueryDTO examQueryDTO, Long userId) {
        PageHelper.startPage(examQueryDTO.getPageNum(), examQueryDTO.getPageSize());
        if (ExamListType.USER_EXAM_LIST.getValue().equals(examQueryDTO.getType())) {
            // 查询用户竞赛列表
            return userExamMapper.selectUserExamList(userId);
        } else {
            // 查询C端的竞赛列表
            return examMapper.selectExamList(examQueryDTO);
        }
    }

    /**
     * 根据竞赛ID列表从 Redis 缓存中获取竞赛的详细信息，并组装成竞赛VO列表。
     *
     * @param examIdList 竞赛ID列表
     * @return 竞赛VO列表。如果 Redis 中没有数据或数据不完整，返回 null。
     */
    private List<ExamVO> assembleExamVOList(List<Long> examIdList) {
        if (CollectionUtil.isEmpty(examIdList)) {
            // 如果竞赛ID列表为空，说明 Redis 中没有数据，返回 null
            return null;
        }

        List<String> detailKeyList = new ArrayList<>();
        for (Long examId : examIdList) {
            // 为每个竞赛ID生成对应的缓存键，并添加到列表中
            detailKeyList.add(getDetailKey(examId));
        }

        // 从 Redis 中批量获取竞赛的详细信息
        List<ExamVO> examVOList = redisService.multiGet(detailKeyList, ExamVO.class);

        // 移除列表中的 null 值
        CollUtil.removeNull(examVOList);

        // 检查获取的竞赛VO列表是否为空或数量不匹配
        if (CollectionUtil.isEmpty(examVOList) || examVOList.size() != examIdList.size()) {
            // 如果 Redis 中数据有问题（如部分数据缺失），返回 null
            return null;
        }

        // 返回组装好的竞赛VO列表
        return examVOList;
    }

    /**
     * 根据竞赛列表类型和用户ID生成缓存键。
     *
     * @param examListType 竞赛列表类型（枚举值）
     * @param userId       用户ID（用于生成用户竞赛列表的缓存键）
     * @return 缓存键
     */
    private String getExamListKey(Integer examListType, Long userId) {
        if (ExamListType.EXAM_UN_FINISH_LIST.getValue().equals(examListType)) {
            // 未完成竞赛列表的缓存键
            return CacheConstants.EXAM_UNFINISHED_LIST;
        } else if (ExamListType.EXAM_HISTORY_LIST.getValue().equals(examListType)) {
            // 历史竞赛列表的缓存键
            return CacheConstants.EXAM_HISTORY_LIST;
        } else {
            // 用户竞赛列表的缓存键，拼接用户ID以确保唯一性
            return CacheConstants.USER_EXAM_LIST + userId;
        }
    }

    /**
     * 获取竞赛详情的缓存键。
     *
     * @param examId 竞赛ID
     * @return 缓存键
     */
    private String getDetailKey(Long examId) {
        return CacheConstants.EXAM_DETAIL + examId;
    }

    /**
     * 获取用户竞赛列表的缓存键。
     *
     * @param userId 用户ID
     * @return 缓存键
     */
    private String getUserExamListKey(Long userId) {
        return CacheConstants.USER_EXAM_LIST + userId;
    }

    /**
     * 获取竞赛题目列表的缓存键。
     *
     * @param examId 竞赛ID
     * @return 缓存键
     */
    private String getExamQuestionListKey(Long examId) {
        return CacheConstants.EXAM_QUESTION_LIST + examId;
    }

    /**
     * 获取竞赛排名列表的缓存键。
     *
     * @param examId 竞赛ID
     * @return 缓存键
     */
    private String getExamRankListKey(Long examId) {
        return CacheConstants.EXAM_RANK_LIST + examId;
    }
}
